package com.yupi.springbootinit.config;

import org.jetbrains.annotations.NotNull;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.ThreadFactory;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;

@Configuration
public class ThreadPoolExcutorConfig {

    @Bean
    public ThreadPoolExecutor threadPoolExecutor(){

        // 自定义线程工厂
        ThreadFactory threadFactory = new ThreadFactory(){
            private int count = 1;

            @Override
            public Thread newThread(@NotNull Runnable r){
                Thread thread = new Thread(r);
                thread.setName("线程" + count);
                count++;
                return thread;
            }
        };

        // 创建线程池，核心线程 2，最大线程 4，非核心线程空闲时间 100s，阻塞队列长 4，自定义线程工厂
        ThreadPoolExecutor threadPoolExecutor = new ThreadPoolExecutor(2, 4, 100,
                TimeUnit.SECONDS, new ArrayBlockingQueue<>(4), threadFactory);

        return threadPoolExecutor;
    }
}
